from envs import SchEnv
from agents import AgentMAPPO_SCH
from train import Config
from train import train_agent


# 指定环境和代理
env = SchEnv
agent = AgentMAPPO_SCH

# 指定超参数
args = Config(agent, env)
args.n_agents = 14
args.action_dim = 2
args.state_dim = 4

# 训练和评估
train_agent(args)
